package TCPUpLoad;

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.nio.charset.StandardCharsets;

/**
 * @Author: ylc
 * @Description:读取客户端上传的文件,保存到服务器的硬盘,给客户端回写"上传成功"
 * @Date Created in 2022-04-28 15:45
 */
public class Server {
    public static void main(String[] args) throws IOException {
        //创建一个服务器ServerSocket对象,和系统要指定的端口号
        ServerSocket server = new ServerSocket(9999);
        while(true){
            //使用ServerSocket对象中的方法accept,获取到请求的客户端Socket对象，也就是等待客户端连接
            Socket client = server.accept();
            new Thread(new Runnable() {
                @Override
                public void run() {
                    try{
                        // 使用Socket对象中的方法getInputStream,获取到网络字节输入流InputStream对象
                        InputStream is = client.getInputStream();
                        // 判断D:\DemoUpload文件夹是否存在,不存在则创建
                        File file = new File("D:\\DemoUpload");
                        if(!file.exists()){
                            file.mkdirs();
                        }

                        String fileName = System.currentTimeMillis() + ".jpg";
                        // 创建一个本地字节输出流FileOutputStream对象,构造方法中绑定要输出的目的地
                        FileOutputStream fos = new FileOutputStream(file + "\\" + fileName);

                        int len = 0;
                        byte[] bytes = new byte[1024];
                        while((len = is.read(bytes)) != -1){
                            fos.write(bytes,0,len);
                        }

                        // 响应客户端
                        client.getOutputStream().write("上传成功".getBytes("utf-8"));
                        // 关闭资源
                        fos.close();
                        client.close();
                    }catch(IOException e){
                        System.out.println(e);
                    }
                }
            }).start();
        }
        //不关闭服务器，持续接收客户端请求
        //server.close();
    }
}
